from monai.networks.nets import UNet
from . import MODELS


@MODELS.register("UNet")
class UNetModel(UNet):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
